package cn.mesmile.juc.assist.countdownlatch;

import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;

/**
 * @author zb
 * @date 2021/8/17 10:38
 * @Description
 */
public class CountDownLatchDemo {

    /**
     *  CountDownLatch 类可以设置一个计数器，然后通过 countDown 方法来进行减 1 的操作，使用 await 方法等待计数器不大于 0，
     *  然后继续执行 await 方法之后的语句。
     * •  CountDownLatch 主要有两个方法，当一个或多个线程调用 await 方法时，这些线程会阻塞
     * •  其它线程调用 countDown 方法会将计数器减 1(调用 countDown 方法的线程不会阻塞)
     * •  当计数器的值变为 0 时，因 await 方法阻塞的线程会被唤醒，继续执行
     */
    public static void main(String[] args) throws InterruptedException {
        int size = 6;
        CountDownLatch countDownLatch = new CountDownLatch(size);
        for (int i = 0; i < size; i++) {
            int finalI = i;
            new Thread(()->{
                System.out.println("开始执行业务："+ finalI);
                try {
                    Thread.sleep(1000);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
                if ("线程3".equals(Thread.currentThread().getName())){
                    // 抛出异常，countDownLatch.countDown() 无法减去 1 ，导致  countDownLatch.await() 阻塞
                    throw new RuntimeException("抛出异常");
                }
                // 执行完成业务减去 1
                countDownLatch.countDown();
            },"线程"+finalI).start();
        }
        System.out.println(">>> 主线程等待子线程执行完成，当前计数："+ countDownLatch.getCount());
        // countDownLatch 为 0 的时候才会执行 await   可以设置超时时间
        countDownLatch.await(6, TimeUnit.SECONDS);
        System.out.println(">>> 子线程执行完成，主线程继续执行");
    }

}
